package com.zjj.algorithm.learning.dmsxl.array;

import java.util.Arrays;

/**
 * 977. 有序数组的平方 简单题
 * <p>
 * 给你一个按 非递减顺序 排序的整数数组 nums，返回 每个数字的平方 组成的新数组，要求也按 非递减顺序 排序。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：nums = [-4,-1,0,3,10]
 * 输出：[0,1,9,16,100]
 * 解释：平方后，数组变为 [16,1,0,9,100]
 * 排序后，数组变为 [0,1,9,16,100]
 * 示例 2：
 * <p>
 * 输入：nums = [-7,-3,2,3,11]
 * 输出：[4,9,9,49,121]
 * <p>
 * <p>
 * 提示：
 * <p>
 * 1 <= nums.length <= 104
 * -104 <= nums[i] <= 104
 * nums 已按 非递减顺序 排序
 * <p>
 * <p>
 * 进阶：
 * <p>
 * 请你设计时间复杂度为 O(n) 的算法解决本问题
 *
 * @author zjj_admin
 * @date 2022/12/21 14:42
 */
public class SquaresOfSortedArray {

    public static void main(String[] args) {
        int[] nums = new int[]{-7, -3, 2, 3, 11};
        int[] res = sortedSquares(nums);
        System.out.println("res = " + Arrays.toString(res));
    }

    /**
     * 使用双指针方法解题
     * 时间
     * 1 ms
     * 击败
     * 100%
     * 内存
     * 43.4 MB
     * 击败
     * 21.72%
     *
     * @param nums
     * @return
     */
    public static int[] sortedSquares(int[] nums) {
        int l = 0;
        int r = nums.length - 1;
        int index = nums.length - 1;
        int[] res = new int[index + 1];
        while (l <= r) {
            int num1 = nums[l] * nums[l];
            int num2 = nums[r] * nums[r];
            if (num1 >= num2) {
                res[index--] = num1;
                l++;
            } else {
                res[index--] = num2;
                r--;
            }
        }
        return res;
    }
}
